Feature engineering for electricity load forecasting#

The purpose of this notebook is to demonstrate how to use skrub and polars to perform feature engineering for electricity load forecasting.

We will build a set of features from different sources:

  • Historical weather data for 10 medium to large urban areas in France;

  • Holidays and calendar features for France;

  • Historical electricity load data for the whole of France.

All these data sources cover a time range from March 23, 2021 to May 31, 2025.

Since our maximum forecasting horizon is 24 hours, we consider that the future weather data is known at a chosen prediction time. Similarly, the holidays and calendar features are known at prediction time for any point in the future.

Therefore, features derived from the weather and calendar data can be used to engineer “future covariates”. Since the load data is our prediction target, we will can also use it to engineer “past covariates” such as lagged features and rolling aggregations.

Environment setup#

We need to install some extra dependencies for this notebook if needed (when running jupyterlite). We need the development version of skrub to be able to use the skrub expressions.

%pip install -q https://pypi.anaconda.org/ogrisel/simple/polars/1.24.0/polars-1.24.0-cp39-abi3-emscripten_3_1_58_wasm32.whl
%pip install -q altair holidays https://pypi.anaconda.org/ogrisel/simple/skrub/0.6.dev0/skrub-0.6.dev0-py3-none-any.whl
ERROR: polars-1.24.0-cp39-abi3-emscripten_3_1_58_wasm32.whl is not a supported wheel on this platform.

Note: you may need to restart the kernel to use updated packages.
Note: you may need to restart the kernel to use updated packages.
# The following 3 imports are only needed to workaround some limitations
# when using polars in a pyodide/jupyterlite notebook.
import tzdata  # noqa: F401
import pandas as pd
from pyarrow.parquet import read_table

import polars as pl
import skrub
from pathlib import Path
import holidays
import warnings

# Ignore warnings from pkg_resources triggered by Python 3.13's multiprocessing.
warnings.filterwarnings("ignore", category=UserWarning, module="pkg_resources")

Time range#

Let’s define a hourly time range from March 23, 2021 to May 31, 2025 that will be used to join the electricity load data and the weather data. The time range is in UTC timezone to avoid any ambiguity when joining with the weather data that is also in UTC.

We wrap the polars dataframe in a skrub variable to benefit from the built-in TableReport display in the notebook. Using the skrub expression system will also be useful later.

time_range_start = pl.datetime(2021, 3, 23, hour=0, time_zone="UTC")
time_range_end = pl.datetime(2025, 5, 31, hour=23, time_zone="UTC")
time = skrub.var(
    "time",
    pl.DataFrame().with_columns(
        pl.datetime_range(
            start=time_range_start,
            end=time_range_end,
            time_zone="UTC",
            interval="1h",
        ).alias("time"),
    ),
)
time
<Var 'time'>
Show graph Var 'time'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

To avoid network issues when running this notebook, the necessary data files have already been downloaded and saved in the datasets folder. See the README.md file for instructions to download the data manually if you want to re-run this notebook with more recent data.

data_source_folder = Path("../datasets")
for data_file in sorted(data_source_folder.iterdir()):
    print(data_file)
../datasets/README.md
../datasets/Total Load - Day Ahead _ Actual_202101010000-202201010000.csv
../datasets/Total Load - Day Ahead _ Actual_202201010000-202301010000.csv
../datasets/Total Load - Day Ahead _ Actual_202301010000-202401010000.csv
../datasets/Total Load - Day Ahead _ Actual_202401010000-202501010000.csv
../datasets/Total Load - Day Ahead _ Actual_202501010000-202601010000.csv
../datasets/weather_bayonne.parquet
../datasets/weather_brest.parquet
../datasets/weather_lille.parquet
../datasets/weather_limoges.parquet
../datasets/weather_lyon.parquet
../datasets/weather_marseille.parquet
../datasets/weather_nantes.parquet
../datasets/weather_paris.parquet
../datasets/weather_strasbourg.parquet
../datasets/weather_toulouse.parquet

List of 10 medium to large urban areas to approximately cover most regions in France with a slight focus on most populated regions that are likely to drive electricity demand.

city_names = [
    "paris",
    "lyon",
    "marseille",
    "toulouse",
    "lille",
    "limoges",
    "nantes",
    "strasbourg",
    "brest",
    "bayonne",
]
all_city_weather_raw = {}
for city_name in city_names:
    # all_city_weather_raw[city_name] = skrub.var(
    # f"{city_name}_weather_raw",
    all_city_weather_raw[city_name] = (
        pl.from_arrow(read_table(f"../datasets/weather_{city_name}.parquet"))
    ).with_columns(
        [
            pl.col("time").dt.cast_time_unit(
                "us"
            ),  # Ensure time column has the same type
        ]
    )
all_city_weather_raw["brest"]
shape: (38_688, 7)
timetemperature_2mprecipitationwind_speed_10mcloud_coversoil_moisture_1_to_3cmrelative_humidity_2m
datetime[μs, UTC]f32f32f32f32f32f32
2021-01-01 00:00:00 UTCnullnullnullnullnullnull
2021-01-01 01:00:00 UTCnullnullnullnullnullnull
2021-01-01 02:00:00 UTCnullnullnullnullnullnull
2021-01-01 03:00:00 UTCnullnullnullnullnullnull
2021-01-01 04:00:00 UTCnullnullnullnullnullnull
2025-05-31 19:00:00 UTC17.51750.012.06945.00.16873.0
2025-05-31 20:00:00 UTC16.26750.09.11447199.00.16877.0
2025-05-31 21:00:00 UTC15.51750.07.55999993.00.16984.0
2025-05-31 22:00:00 UTC15.56750.09.0100.00.1782.0
2025-05-31 23:00:00 UTC15.56750.05.506941100.00.17181.0
all_city_weather_raw["brest"].drop_nulls(subset=["temperature_2m"])
shape: (36_744, 7)
timetemperature_2mprecipitationwind_speed_10mcloud_coversoil_moisture_1_to_3cmrelative_humidity_2m
datetime[μs, UTC]f32f32f32f32f32f32
2021-03-23 00:00:00 UTC4.628null10.086427nullnull94.0
2021-03-23 01:00:00 UTC5.0280.011.1832016.0null95.0
2021-03-23 02:00:00 UTC5.0780.010.9667136.0null94.0
2021-03-23 03:00:00 UTC4.6280.010.4647975.0null93.0
2021-03-23 04:00:00 UTC4.4280.010.4647975.0null92.0
2025-05-31 19:00:00 UTC17.51750.012.06945.00.16873.0
2025-05-31 20:00:00 UTC16.26750.09.11447199.00.16877.0
2025-05-31 21:00:00 UTC15.51750.07.55999993.00.16984.0
2025-05-31 22:00:00 UTC15.56750.09.0100.00.1782.0
2025-05-31 23:00:00 UTC15.56750.05.506941100.00.17181.0
all_city_weather = time.skb.eval()
for city_name, city_weather_raw in all_city_weather_raw.items():
    all_city_weather = all_city_weather.join(
        city_weather_raw.rename(lambda x: x if x == "time" else x + "_" + city_name),
        on="time",
        how="inner",
    )

all_city_weather = skrub.var(
    "all_city_weather",
    all_city_weather,
)
all_city_weather
<Var 'all_city_weather'>
Show graph Var 'all_city_weather'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

Calendar and holidays features#

We leverage the holidays package to enrich the time range with some calendar features such as public holidays in France. We also add some features that are useful for time series forecasting such as the day of the week, the day of the year, and the hour of the day.

Note that the holidays package requires us to extract the date for the French timezone.

Similarly for the calendar features: all the time features are extracted from the time in the French timezone.

holidays_fr = holidays.country_holidays("FR", years=range(2019, 2026))

fr_time = pl.col("time").dt.convert_time_zone("Europe/Paris")
calendar = time.with_columns(
    [
        fr_time.dt.date().is_in(holidays_fr.keys()).alias("is_holiday_fr"),
        fr_time.dt.weekday().alias("day_of_week_fr"),
        fr_time.dt.ordinal_day().alias("day_of_year_fr"),
        fr_time.dt.hour().alias("hour_of_day_fr"),
    ],
)
calendar
<CallMethod 'with_columns'>
Show graph Var 'time' CallMethod 'with_columns'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

Electricity load data#

Finally we load the electricity load data. This data will both be used as a target variable but also to craft some lagged and window-aggregated features.

load_data_files = [
    data_file
    for data_file in sorted(data_source_folder.iterdir())
    if data_file.name.startswith("Total Load - Day Ahead")
    and data_file.name.endswith(".csv")
]
electricity_raw = skrub.var(
    "electricity_raw",
    pl.concat(
        [
            pl.from_pandas(pd.read_csv(data_file, na_values=["N/A", "-"])).drop(
                ["Day-ahead Total Load Forecast [MW] - BZN|FR"]
            )
            for data_file in load_data_files
        ],
        how="vertical",
    ),
)
electricity_raw
<Var 'electricity_raw'>
Show graph Var 'electricity_raw'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

electricity = (
    electricity_raw.with_columns(
        [
            pl.col("Time (UTC)")
            .str.split(by=" - ")
            .list.first()
            .str.to_datetime("%d.%m.%Y %H:%M", time_zone="UTC")
            .alias("time"),
        ]
    )
    .drop(["Time (UTC)"])
    .rename({"Actual Total Load [MW] - BZN|FR": "load_mw"})
    .filter(pl.col("time").dt.minute().eq(0))
    .filter(pl.col("time") >= time_range_start)
    .filter(pl.col("time") <= time_range_end)
    .select(["time", "load_mw"])
)
electricity
<CallMethod 'select'>
Show graph Var 'electricity_raw' CallMethod 'with_columns' CallMethod 'drop' CallMethod 'rename' CallMethod 'filter' CallMethod 'filter' CallMethod 'filter' CallMethod 'select'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

electricity.filter(pl.col("load_mw").is_null())
<CallMethod 'filter'>
Show graph Var 'electricity_raw' CallMethod 'with_columns' CallMethod 'drop' CallMethod 'rename' CallMethod 'filter' CallMethod 'filter' CallMethod 'filter' CallMethod 'select' CallMethod 'filter'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

electricity.filter(
    (pl.col("time") > pl.datetime(2021, 10, 30, hour=10, time_zone="UTC"))
    & (pl.col("time") < pl.datetime(2021, 10, 31, hour=10, time_zone="UTC"))
).skb.eval().plot.line(x="time:T", y="load_mw:Q")
electricity = electricity.with_columns([pl.col("load_mw").interpolate()])
electricity.filter(
    (pl.col("time") > pl.datetime(2021, 10, 30, hour=10, time_zone="UTC"))
    & (pl.col("time") < pl.datetime(2021, 10, 31, hour=10, time_zone="UTC"))
).skb.eval().plot.line(x="time:T", y="load_mw:Q")

Check that the number of rows matches our expectations based on the number of hours that separate the first and the last dates. We can do that by joining with the time range dataframe and checking that the number of rows stays the same.

assert (
    time.join(electricity, on="time", how="inner").shape[0] == time.shape[0]
).skb.eval()

Lagged features#

We can now create some lagged features from the electricity load data.

We will create 3 hourly lagged features, 1 daily lagged feature, and 1 weekly lagged feature. We will also create a rolling median and inter-quartile feature over the last 24 hours and over the last 7 days.

def iqr(col, *, window_size: int):
    """Inter-quartile range (IQR) of a column."""
    return col.rolling_quantile(0.75, window_size=window_size) - col.rolling_quantile(
        0.25, window_size=window_size
    )


electricity_lagged = electricity.with_columns(
    [pl.col("load_mw").shift(i).alias(f"load_mw_lag_{i}h") for i in range(1, 4)]
    + [
        pl.col("load_mw").shift(24).alias("load_mw_lag_1d"),
        pl.col("load_mw").shift(24 * 7).alias("load_mw_lag_1w"),
        pl.col("load_mw")
        .rolling_median(window_size=24)
        .alias("load_mw_rolling_median_24h"),
        pl.col("load_mw")
        .rolling_median(window_size=24 * 7)
        .alias("load_mw_rolling_median_7d"),
        iqr(pl.col("load_mw"), window_size=24).alias("load_mw_iqr_24h"),
        iqr(pl.col("load_mw"), window_size=24 * 7).alias("load_mw_iqr_7d"),
    ],
)
electricity_lagged
<CallMethod 'with_columns'>
Show graph Var 'electricity_raw' CallMethod 'with_columns' CallMethod 'drop' CallMethod 'rename' CallMethod 'filter' CallMethod 'filter' CallMethod 'filter' CallMethod 'select' CallMethod 'with_columns' CallMethod 'with_columns'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

import altair


altair.Chart(electricity_lagged.tail(100).skb.eval()).transform_fold(
    [
        "load_mw",
        "load_mw_lag_1h",
        "load_mw_lag_2h",
        "load_mw_lag_3h",
        "load_mw_lag_1d",
        "load_mw_lag_1w",
        "load_mw_rolling_median_24h",
        "load_mw_rolling_median_7d",
        "load_mw_iqr_24h",
        "load_mw_iqr_7d",
    ],
    as_=["key", "load_mw"],
).mark_line(tooltip=True).encode(x="time:T", y="load_mw:Q", color="key:N").interactive()

Investigating outliers in the lagged features#

Let’s use the skrub.TableReport tool to look at the plots of the marginal distribution of the lagged features.

from skrub import TableReport

TableReport(electricity_lagged.skb.eval())
Processing column   1 / 11
Processing column   2 / 11
Processing column   3 / 11
Processing column   4 / 11
Processing column   5 / 11
Processing column   6 / 11
Processing column   7 / 11
Processing column   8 / 11
Processing column   9 / 11
Processing column  10 / 11
Processing column  11 / 11

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

Let’s extract the dates where the inter-quartile range of the load is greater than 15,000 MW.

electricity_lagged.filter(pl.col("load_mw_iqr_7d") > 15_000)[
    "time"
].dt.date().unique().sort().to_list().skb.eval()
[datetime.date(2021, 12, 26),
 datetime.date(2021, 12, 27),
 datetime.date(2021, 12, 28),
 datetime.date(2022, 1, 7),
 datetime.date(2022, 1, 8),
 datetime.date(2023, 1, 19),
 datetime.date(2023, 1, 20),
 datetime.date(2023, 1, 21),
 datetime.date(2024, 1, 10),
 datetime.date(2024, 1, 11),
 datetime.date(2024, 1, 12),
 datetime.date(2024, 1, 13)]

We observe 3 date ranges with high inter-quartile range. Let’s plot the electricity load and the lagged features for the first data range along with the weather data for Paris.

altair.Chart(
    electricity_lagged.filter(
        (pl.col("time") > pl.datetime(2021, 12, 1, time_zone="UTC"))
        & (pl.col("time") < pl.datetime(2021, 12, 31, time_zone="UTC"))
    ).skb.eval()
).transform_fold(
    [
        "load_mw",
        "load_mw_iqr_7d",
    ],
).mark_line(
    tooltip=True
).encode(
    x="time:T", y="value:Q", color="key:N"
).interactive()
altair.Chart(
    all_city_weather.filter(
        (pl.col("time") > pl.datetime(2021, 12, 1, time_zone="UTC"))
        & (pl.col("time") < pl.datetime(2021, 12, 31, time_zone="UTC"))
    ).skb.eval()
).transform_fold(
    [f"temperature_2m_{city_name}" for city_name in city_names],
).mark_line(
    tooltip=True
).encode(
    x="time:T", y="value:Q", color="key:N"
).interactive()

Based on the plots above, we can see that the electricity load was high just before the Christmas holidays due to low temperatures. Then the load suddenly dropped because temperatures went higher right at the start of the end-of-year holidays.

So those outliers do not seem to be caused to a data quality issue but rather due to a real change in the electricity load demand. We could conduct similar analysis for the other date ranges with high inter-quartile range but we will skip that for now.

If we had observed significant data quality issues over extended periods of time could have been addressed by removing the corresponding rows from the dataset. However, this would make the lagged and windowing feature engineering challenging to reimplement correctly. A better approach would be to keep a contiguous dataset assign 0 weights to the affected rows when fitting or evaluating the trained models via the use of the sample_weight parameter.

Final dataset#

We now assemble the dataset that will be used to train and evaluate the forecasting models via backtesting.

prediction_time = time = skrub.var(
    "prediction_time",
    pl.DataFrame().with_columns(
        pl.datetime_range(
            start=time_range_start + pl.duration(days=7),
            end=time_range_end - pl.duration(hours=24),
            time_zone="UTC",
            interval="1h",
        ).alias("prediction_time"),
    ),
)
prediction_time
<Var 'prediction_time'>
Show graph Var 'prediction_time'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

features = (
    (
        prediction_time.join(
            electricity_lagged, left_on="prediction_time", right_on="time"
        )
        .join(all_city_weather, left_on="prediction_time", right_on="time")
        .join(calendar, left_on="prediction_time", right_on="time")
    )
    .drop("prediction_time")
    .skb.mark_as_X()
)
features
<CallMethod 'drop'>
Show graph Var 'prediction_time' CallMethod 'join' Var 'electricity_raw' CallMethod 'with_columns' CallMethod 'drop' CallMethod 'rename' CallMethod 'filter' CallMethod 'filter' CallMethod 'filter' CallMethod 'select' CallMethod 'with_columns' CallMethod 'with_columns' CallMethod 'join' Var 'all_city_weather' CallMethod 'join' Var 'time' CallMethod 'with_columns' X: CallMethod 'drop'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

horizons = range(1, 3)  # Forecasting horizons from 1 to 24 hours
target_column_name_pattern = "load_mw_horizon_{horizon}h"

targets = prediction_time.join(
    electricity.with_columns(
        [
            pl.col("load_mw")
            .shift(-h)
            .alias(target_column_name_pattern.format(horizon=h))
            for h in horizons
        ]
    ),
    left_on="prediction_time",
    right_on="time",
)
horizon_of_interest = 2
target_column_name = target_column_name_pattern.format(horizon=horizon_of_interest)
predicted_target_column_name = "predicted_" + target_column_name
target = targets[target_column_name].skb.mark_as_y()
from sklearn.ensemble import HistGradientBoostingRegressor


predictions = features.skb.apply(
    HistGradientBoostingRegressor(
        random_state=0,
        learning_rate=skrub.choose_float(
            0.01, 0.9, default=0.1, log=True, name="learning_rate"
        ),
        max_leaf_nodes=skrub.choose_int(
            3, 300, default=30, log=True, name="max_leaf_nodes"
        ),
    ),
    y=target,
)
predictions
<Apply HistGradientBoostingRegressor>
Show graph Var 'prediction_time' CallMethod 'join' CallMethod 'join' Var 'electricity_raw' CallMethod 'with_columns' CallMethod 'drop' CallMethod 'rename' CallMethod 'filter' CallMethod 'filter' CallMethod 'filter' CallMethod 'select' CallMethod 'with_columns' CallMethod 'with_columns' CallMethod 'with_columns' CallMethod 'join' Var 'all_city_weather' CallMethod 'join' Var 'time' CallMethod 'with_columns' X: CallMethod 'drop' Apply HistGradientBoostingRegressor y: GetItem 'load_mw_horizon_2h'

Result:

Please enable javascript

The skrub table reports need javascript to display correctly. If you are displaying a report in a Jupyter notebook and you see this message, you may need to re-execute the cell or to trust the notebook (button on the top right or "File > Trust notebook").

altair.Chart(
    pl.concat(
        [
            targets.skb.eval(),
            predictions.rename(
                {target_column_name: predicted_target_column_name}
            ).skb.eval(),
        ],
        how="horizontal",
    ).tail(24 * 7)
).transform_fold(
    [target_column_name, predicted_target_column_name],
).mark_line(
    tooltip=True
).encode(
    x="prediction_time:T", y="value:Q", color="key:N"
).interactive()
from sklearn.model_selection import TimeSeriesSplit
from sklearn.metrics import make_scorer, mean_absolute_percentage_error, get_scorer

mape_scorer = make_scorer(mean_absolute_percentage_error)

ts_cv_5 = TimeSeriesSplit(n_splits=5, max_train_size=10_000, gap=24)

predictions.skb.cross_validate(
    cv=ts_cv_5,
    scoring={
        "r2": get_scorer("r2"),
        "mape": mape_scorer,
    },
    return_train_score=True,
    verbose=1,
    n_jobs=-1,
).round(3)
[Parallel(n_jobs=-1)]: Using backend LokyBackend with 4 concurrent workers.
[Parallel(n_jobs=-1)]: Done   5 out of   5 | elapsed:    5.4s finished
fit_time score_time test_r2 train_r2 test_mape train_mape
0 1.086 0.066 0.970 0.997 0.021 0.008
1 1.364 0.067 0.983 0.996 0.020 0.010
2 1.432 0.053 0.972 0.995 0.025 0.010
3 1.399 0.066 0.973 0.994 0.023 0.012
4 1.058 0.038 0.977 0.993 0.023 0.014
ts_cv_3 = TimeSeriesSplit(n_splits=3, max_train_size=10_000, gap=24)
randomized_search = predictions.skb.get_randomized_search(
    cv=ts_cv_3,
    scoring="r2",
    n_iter=30,
    fitted=True,
    verbose=1,
    n_jobs=-1,
)
randomized_search.results_
Fitting 3 folds for each of 30 candidates, totalling 90 fits
learning_rate max_leaf_nodes mean_test_score
0 0.101497 126 0.980176
1 0.079763 40 0.980106
2 0.074376 185 0.979518
3 0.056106 231 0.978974
4 0.158866 187 0.978394
5 0.177116 12 0.977895
6 0.181058 8 0.977489
7 0.288475 18 0.976352
8 0.343255 20 0.976326
9 0.102882 8 0.976052
10 0.339992 39 0.975655
11 0.397618 7 0.975605
12 0.308953 224 0.975359
13 0.186020 5 0.974518
14 0.448934 6 0.973496
15 0.370166 4 0.973071
16 0.617871 4 0.972331
17 0.480384 17 0.971334
18 0.508125 14 0.971045
19 0.542180 207 0.969476
20 0.588342 84 0.968365
21 0.204359 3 0.966061
22 0.684306 24 0.962164
23 0.889901 11 0.960339
24 0.045399 5 0.950639
25 0.879902 87 0.949598
26 0.051006 3 0.934190
27 0.022310 8 0.918392
28 0.014183 186 0.910541
29 0.013296 228 0.898933
randomized_search.plot_results()
# nested_cv_results = skrub.cross_validate(
#     environment=predictions.skb.get_data(),
#     pipeline=randomized_search,
#     cv=ts_cv_5,
#     scoring={
#         "r2": get_scorer("r2"),
#         "mape": mape_scorer,
#     },
#     n_jobs=-1,
#     return_pipeline=True,
# ).round(3)
# nested_cv_results
# for outer_cv_idx in range(len(nested_cv_results)):
#     print(
#         nested_cv_results.loc[outer_cv_idx, "pipeline"]
#         .results_.loc[0]
#         .round(3)
#         .to_dict()
#     )
# from joblib import Parallel, delayed

# cv_predictions = []
# for ts_cv_train_idx, ts_cv_test_idx in ts_cv_5.split(prediction_time.skb.eval()):
#     features[ts_cv_train_idx].fit
predictions = features.skb.apply(
    skrub.SelectCols(
        cols=skrub.choose_from(
            [
                skrub.selectors.all(),
                skrub.selectors.filter_names(
                    lambda name: name.startswith("load_mw_"),
                ),
                skrub.selectors.filter_names(
                    lambda name: name.startswith("load_mw_")
                    or name.startswith("temperature_"),
                ),
                skrub.selectors.filter_names(
                    lambda name: name.startswith("load_mw_")
                    or name.startswith("precipitation_"),
                ),
                skrub.selectors.filter_names(
                    lambda name: name.startswith("load_mw_")
                    or name.startswith("wind_speed_"),
                ),
                skrub.selectors.filter_names(
                    lambda name: name.startswith("load_mw_")
                    or name.startswith("cloud_cover_"),
                ),
                # calendar features
                skrub.selectors.filter_names(
                    lambda name: name.startswith("load_mw_") or name.endswith("_fr"),
                ),
            ],
            name="feature_subset",
        )
    )
).skb.apply(
    HistGradientBoostingRegressor(
        random_state=0,
        learning_rate=skrub.choose_float(
            0.01, 0.9, default=0.1, log=True, name="learning_rate"
        ),
    ),
    y=target,
)
randomized_search = predictions.skb.get_randomized_search(
    cv=ts_cv_3,
    scoring="r2",
    n_iter=30,
    fitted=True,
    verbose=1,
    n_jobs=-1,
)
randomized_search.results_
Fitting 3 folds for each of 30 candidates, totalling 90 fits
feature_subset learning_rate mean_test_score
0 all() 0.099536 0.980097
1 all() 0.067589 0.979463
2 filter_names(<lambda>) 0.078805 0.969532
3 filter_names(<lambda>) 0.218262 0.968531
4 filter_names(<lambda>) 0.399262 0.962610
5 all() 0.026653 0.962475
6 filter_names(<lambda>) 0.036151 0.961864
7 filter_names(<lambda>) 0.667629 0.952211
8 filter_names(<lambda>) 0.019793 0.926310
9 filter_names(<lambda>) 0.176460 0.902561
10 filter_names(<lambda>) 0.171368 0.902167
11 filter_names(<lambda>) 0.088878 0.900799
12 filter_names(<lambda>) 0.052764 0.900393
13 filter_names(<lambda>) 0.115793 0.899440
14 filter_names(<lambda>) 0.048053 0.898641
15 filter_names(<lambda>) 0.040954 0.898554
16 filter_names(<lambda>) 0.117140 0.898435
17 filter_names(<lambda>) 0.037720 0.897170
18 filter_names(<lambda>) 0.037271 0.895901
19 filter_names(<lambda>) 0.317319 0.894193
20 filter_names(<lambda>) 0.030575 0.893030
21 filter_names(<lambda>) 0.338578 0.891036
22 filter_names(<lambda>) 0.025810 0.889244
23 filter_names(<lambda>) 0.370924 0.889087
24 filter_names(<lambda>) 0.025857 0.886896
25 filter_names(<lambda>) 0.348733 0.884695
26 filter_names(<lambda>) 0.500632 0.880784
27 filter_names(<lambda>) 0.464976 0.874003
28 filter_names(<lambda>) 0.013596 0.866150
29 filter_names(<lambda>) 0.805895 0.833898
# nested_cv_results = skrub.cross_validate(
#     environment=predictions.skb.get_data(),
#     pipeline=randomized_search,
#     cv=ts_cv_5,
#     scoring={
#         "r2": get_scorer("r2"),
#         "mape": mape_scorer,
#     },
#     n_jobs=-1,
#     return_pipeline=True,
# ).round(3)
# nested_cv_results
targets = targets.skb.drop(cols=["load_mw"]).skb.mark_as_y()
from sklearn.multioutput import MultiOutputRegressor

model = MultiOutputRegressor(
    estimator=HistGradientBoostingRegressor(
        random_state=0,
        learning_rate=skrub.choose_float(
            0.01, 0.9, default=0.1, log=True, name="learning_rate"
        ),
    ),
)
predictions = features.skb.apply(model, y=targets)
from sklearn.metrics import r2_score


def multioutput_scorer(regressor, X, y, score_func, score_name):
    y_pred = regressor.predict(X)
    return {
        f"{score_name}_horizon_{h}h": score
        for h, score in enumerate(
            score_func(y, y_pred, multioutput="raw_values"), start=1
        )
    }


def scoring(regressor, X, y):
    return {
        **multioutput_scorer(regressor, X, y, mean_absolute_percentage_error, "mape"),
        **multioutput_scorer(regressor, X, y, r2_score, "r2"),
    }


predictions.skb.cross_validate(
    cv=ts_cv_5,
    scoring=scoring,
    return_train_score=True,
    verbose=1,
    n_jobs=-1,
).round(3)
[Parallel(n_jobs=-1)]: Using backend LokyBackend with 4 concurrent workers.
[Parallel(n_jobs=-1)]: Done   5 out of   5 | elapsed:    7.6s finished
fit_time score_time test_mape_horizon_1h train_mape_horizon_1h test_mape_horizon_2h train_mape_horizon_2h test_mape_horizon_3h train_mape_horizon_3h test_r2_horizon_1h train_r2_horizon_1h test_r2_horizon_2h train_r2_horizon_2h test_r2_horizon_3h train_r2_horizon_3h
0 3.312 0.318 0.017 0.0 0.013 0.006 0.020 0.008 -19.596 1.000 0.988 0.998 0.974 0.997
1 3.869 0.322 0.019 0.0 0.012 0.007 0.020 0.010 -24.166 0.998 0.993 0.998 0.983 0.996
2 4.326 0.308 0.010 0.0 0.016 0.007 0.025 0.010 -8.691 1.000 0.987 0.998 0.973 0.995
3 3.975 0.332 0.018 0.0 0.015 0.009 0.023 0.012 -23.756 0.999 0.988 0.997 0.973 0.995
4 2.868 0.185 0.018 0.0 0.014 0.009 0.023 0.013 -23.643 0.999 0.991 0.997 0.977 0.993